# -*- coding: utf-8 -*-
# @Time    : 2019/6/2 16:54
# @Author  : DrMa

class Our_model_config_CAIL(object):
    def __init__(self):
        self.num_label=106+2
        self.doc_len=400
        self.learning_rate = 0.001
        self.dropout_keep_prob = 0.5
        self.rnn_type = 'lstm'
        self.hidden_size=50
        self.batch_size=32
        self.num_layers=2
        self.num_sentence=40
        self.embedding_dim=100
        self.threshold_value=0.3

        self.sos_label_id=107
        self.eos_label_id=106
        self.max_label_num=3
        self.embedding_dim_label=100
class Our_model_config_predict_CAIL(object):
    def __init__(self):
        self.num_label=106+2
        self.doc_len=400
        self.learning_rate = 0.001
        self.dropout_keep_prob = 1.0
        self.rnn_type = 'lstm'
        self.hidden_size=50
        self.batch_size=1
        self.num_layers=2
        self.num_sentence=40
        self.embedding_dim=100
        self.threshold_value=0.3

        self.sos_label_id = 107
        self.eos_label_id = 106
        self.max_label_num = 5
        self.embedding_dim_label = 100

class Our_model_config_CJO(object):
    def __init__(self):
        self.num_label=131+2
        self.sos_label_id = 132
        self.eos_label_id = 131
        self.max_label_num = 3

        self.learning_rate = 0.001
        self.dropout_keep_prob = 0.5

        self.rnn_type = 'lstm'
        self.hidden_size=50
        self.num_layers=2

        self.len_doc = 300
        self.len_law = 500
        self.num_sentence_fact=30
        self.num_sentence_law=20

        self.embedding_dim_word=100
        self.embedding_dim_label = 100

        self.threshold_value=0.3
        self.batch_size = 32
class Our_model_config_predict_CJO(object):
    def __init__(self):
        self.num_label = 131 + 2
        self.sos_label_id = 132
        self.eos_label_id = 131
        self.max_label_num = 3

        self.learning_rate = 0.001
        self.dropout_keep_prob = 1.0

        self.rnn_type = 'lstm'
        self.hidden_size = 50
        self.num_layers = 2

        self.len_doc = 300
        self.len_law = 500
        self.num_sentence_fact = 30
        self.num_sentence_law = 20

        self.embedding_dim_word = 100
        self.embedding_dim_label = 100

        self.threshold_value = 0.3
        self.batch_size = 1